from setuptools import setup, find_packages
import os
from glob import glob

package_name = 'lane_detection'

setup(
    name=package_name,
    version='0.0.0',
    packages=find_packages(exclude=['test']),  # 自动查找所有包
    data_files=[
        ('share/ament_index/resource_index/packages',
            ['resource/' + package_name]),
        ('share/' + package_name, ['package.xml']),
        (os.path.join('share', package_name, 'launch'), glob('launch/*.launch.py')),
        (os.path.join('share', package_name, 'config'), glob('config/*.yaml')),
        (os.path.join('share', package_name, 'models'), glob('models/*.pt')),
        (os.path.join('share', package_name, 'resource'), glob('resource/*')),  # 添加资源文件
    ],
    install_requires=['setuptools'],
    zip_safe=True,
    maintainer='RockyFang',
    maintainer_email='2606776129@qq.com',
    description='Lane detection package using YOLO and BEV transformation',
    license='Apache License 2.0',
    tests_require=['pytest'],
    entry_points={
        'console_scripts': [
            'lane_detection= lane_detection.lane_detection:main',
            'camera_ori_node = lane_detection.camera_ori_node:main',
        ],
    },
)